package com.book.service;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.SocketException;

import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;
import org.apache.log4j.Logger;
import org.springframework.stereotype.Component;

import com.book.constant.Constants;

@Component
public class FtpService {

	private static final Logger logger = Logger.getLogger(FtpService.class);

	/**
	 * Description: 向FTP服务器上传文件
	 * @param url FTP服务器hostname
	 * @param port FTP服务器端口
	 * @param username FTP登录账号
	 * @param password FTP登录密码
	 * @param path FTP服务器保存目录
	 * @param filename 上传到FTP服务器上的文件名
	 * @param input 输入流
	 * @return 成功返回true，否则返回false
	 * @throws IOException 
	 */
	public static boolean uploadFile(String url, int port, String username, String password, String path, String filename, InputStream input) throws IOException {
		boolean success = false;
		FTPClient ftp = new FTPClient();
		ftp.setControlEncoding(Constants.FTP_ENCODE);

		try {
			int reply;
			ftp.connect(url, port);//连接FTP服务器
			//如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
			ftp.login(username, password);//登录
			ftp.setFileType(FTPClient.BINARY_FILE_TYPE);
			reply = ftp.getReplyCode();
			if (!FTPReply.isPositiveCompletion(reply)) {
				ftp.disconnect();
				return success;
			}
			ftp.makeDirectory(path);
			ftp.changeWorkingDirectory(path);
			long begin = System.currentTimeMillis();
			logger.info(path + ":" + filename + "===>" + ftp.storeFile(filename, input));
			logger.info("upload cost " + (System.currentTimeMillis() - begin) / 1000 + " s");

			input.close();
			ftp.logout();
			success = true;
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			if (ftp.isConnected()) {
				try {
					ftp.disconnect();
				} catch (IOException ioe) {
				}
			}
		}
		return success;
	}

	/**
	 * Description: 从FTP服务器下载文件
	 * @param url FTP服务器hostname
	 * @param port FTP服务器端口
	 * @param username FTP登录账号
	 * @param password FTP登录密码
	 * @param remotePath FTP服务器上的相对路径
	 * @param fileName 要下载的文件名
	 * @param localPath 下载后保存到本地的路径
	 * @return
	 */
	public static boolean downFile2(String url, int port, String username, String password, String remotePath, String fileName, OutputStream os) {
		boolean success = false;
		FTPClient ftp = new FTPClient();
		ftp.setControlEncoding(Constants.FTP_ENCODE);
		try {
			int reply;
			ftp.connect(url, port);
			//如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
			ftp.login(username, password);//登录
			ftp.setFileType(FTPClient.BINARY_FILE_TYPE);
			reply = ftp.getReplyCode();
			if (!FTPReply.isPositiveCompletion(reply)) {
				ftp.disconnect();
				return success;
			}
			ftp.changeWorkingDirectory(remotePath);//转移到FTP服务器目录
			FTPFile[] fs = ftp.listFiles();
			for (FTPFile ff : fs) {
				if (ff.getName().equals(fileName)) {
					InputStream retrieveFileStream = ftp.retrieveFileStream(ff.getName());
					FileEncryptGenService.change(retrieveFileStream, os);
				}
			}
			ftp.logout();
			success = true;
			logger.info(remotePath + ":" + fileName + " download ===> " + success);
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			if (ftp.isConnected()) {
				try {
					ftp.disconnect();
				} catch (IOException ioe) {
				}
			}
		}
		return success;
	}

	/**
	 * Description: 从FTP服务器下载文件
	 * @param url FTP服务器hostname
	 * @param port FTP服务器端口
	 * @param username FTP登录账号
	 * @param password FTP登录密码
	 * @param remotePath FTP服务器上的相对路径
	 * @param fileName 要下载的文件名
	 * @param localPath 下载后保存到本地的路径
	 * @return
	 */
	public static boolean downFile(String url, int port, String username, String password, String remotePath, String fileName, OutputStream os) {
		boolean success = false;
		FTPClient ftp = new FTPClient();
		ftp.setControlEncoding(Constants.FTP_ENCODE);
		try {
			int reply;
			ftp.connect(url, port);
			//如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
			ftp.login(username, password);//登录
			ftp.setFileType(FTPClient.BINARY_FILE_TYPE);
			reply = ftp.getReplyCode();
			if (!FTPReply.isPositiveCompletion(reply)) {
				ftp.disconnect();
				return success;
			}
			ftp.changeWorkingDirectory(remotePath);//转移到FTP服务器目录

			FTPFile[] fs = ftp.listFiles();
			for (FTPFile ff : fs) {
				if (ff.getName().equals(fileName)) {
					InputStream retrieveFileStream = ftp.retrieveFileStream(ff.getName());
					FileEncryptGenService.change(retrieveFileStream, os);
					if (retrieveFileStream != null) {
						retrieveFileStream.close();
					}
					if (!ftp.completePendingCommand()) {
						ftp.logout();
						ftp.disconnect();
					}
				}
			}
			success = true;
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			if (ftp.isConnected()) {
				try {
					ftp.disconnect();
				} catch (IOException ioe) {
				}
			}
		}
		return success;
	}

	public static void delFtpFile(String url, int port, String username, String password, String remotePath, String filePath) throws SocketException, IOException {
		FTPClient ftp = new FTPClient();
		ftp.setControlEncoding(Constants.FTP_ENCODE);
		int reply;
		ftp.connect(url, port);
		//如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
		ftp.login(username, password);//登录
		reply = ftp.getReplyCode();
		if (!FTPReply.isPositiveCompletion(reply)) {
			ftp.disconnect();
		}
		ftp.changeWorkingDirectory(remotePath);//转移到FTP服务器目录
		boolean result = ftp.deleteFile(filePath);
		logger.info(remotePath + ":" + filePath + " delete ===> " + result);
	}

	/**
	 * Description: 从FTP服务器下载文件
	 * @param url FTP服务器hostname
	 * @param port FTP服务器端口
	 * @param username FTP登录账号
	 * @param password FTP登录密码
	 * @param remotePath FTP服务器上的相对路径
	 * @param fileName 要下载的文件名
	 * @param localPath 下载后保存到本地的路径
	 * @return
	 */
	public static boolean showFileContent(String url, int port, String username, String password, String remotePath, String fileName, OutputStream outputStream) {
		boolean success = false;
		FTPClient ftp = new FTPClient();
		ftp.setControlEncoding(Constants.FTP_ENCODE);
		try {
			int reply;
			ftp.connect(url, port);
			//如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
			ftp.login(username, password);//登录
			reply = ftp.getReplyCode();
			if (!FTPReply.isPositiveCompletion(reply)) {
				ftp.disconnect();
				return success;
			}
			ftp.changeWorkingDirectory(remotePath);//转移到FTP服务器目录
			FTPFile[] fs = ftp.listFiles();
			for (FTPFile ff : fs) {
				if (ff.getName().equals(fileName)) {
					InputStream retrieveFileStream = ftp.retrieveFileStream(ff.getName());
					FileEncryptGenService.change(retrieveFileStream, outputStream);
				}
			}
			ftp.logout();
			success = true;
			logger.info(remotePath + ":" + fileName + " showContent ===> " + success);
		} catch (IOException e) {
			e.printStackTrace();
		} finally {
			if (ftp.isConnected()) {
				try {
					ftp.disconnect();
				} catch (IOException ioe) {
				}
			}
		}
		return success;
	}

}
